#ifndef __M_BUFFER_H__
#define __M_BUFFER_H__

#include <vector>
#include <iostream>
#include <cassert>
namespace log
{
#define DEFAULT_BUFFER_SIZE (1 * 1024 * 1024)
#define THRESHOLD_SIZE (8 * 1024 * 1024)        // 阈值 扩容时使用小于该值翻倍增长
#define INCREMENT_BUFFER_SIZE (1 * 1024 * 1024) // 增量 空间大小线性增长
    class Buffer
    {
    public:
        Buffer() : _buffer(DEFAULT_BUFFER_SIZE), _reader_idx(0), _write_idx(0)
        {
        }
        // 向缓冲区写入数据
        void push(const char *data, size_t len)
        {
            // 1. 缓冲区剩余空间不够的情况：1.扩容 2. 阻塞/返回false
            // 固定大小，直接返回
            // if (len > writeableSize())
            //     return;
            // 动态空间，用于极限性能测试--扩容
            ensureEnoughSize(len);
            // 2.将数据拷贝到缓冲区
            std::copy(data, data + len, &_buffer[_write_idx]); // 拷贝函数 起始位置 终止位置 放入位置
            // 3.将当前写入位置向后偏移
            moveWrite(len);
        }
        void moveReader(size_t len)
        {
            assert(len <= readableSize()); // 小于可读数据长度
            _reader_idx += len;
        }
        // 返回可读数据的起始地址
        const char *begin()
        {
            return &_buffer[_reader_idx];
        }
        // 返回可读数据的长度
        size_t readableSize()
        {
            return (_write_idx - _reader_idx);
        }
        // 返回可写数据的长度
        size_t writeableSize()
        {
            // 该接口针对固定大小缓冲区，扩容思路不存在可写空间大小，因为总是可写的
            return (_buffer.size() - _write_idx);
        }
        // 重置读写位置,初始化缓冲区
        void reset()
        {
            _reader_idx = 0;
            _write_idx = 0;
        }
        // 对buffer实现交换操作
        void swap(Buffer &buffer)
        {
            _buffer.swap(buffer._buffer);
            std::swap(_reader_idx, buffer._reader_idx);
            std::swap(_write_idx, buffer._write_idx);
        }
        // 判断缓冲区是否为空
        bool empty()
        {
            return (_reader_idx == _write_idx);
        }

    private:
        // 对空间进行扩容
        void ensureEnoughSize(size_t len)
        {
            if (len < writeableSize())
                return; // 不需要扩容
            size_t new_size = 0;
            if (_buffer.size() < THRESHOLD_SIZE)
            {
                new_size = _buffer.size() * 2 + len; // + len 防止*2 后仍然大小不够
            }
            else
            {
                new_size = _buffer.size() + INCREMENT_BUFFER_SIZE +len; // + len 防止线性增长 后仍然大小不够
            }
            _buffer.resize(new_size);
        }
        // 对读写指针进行向后偏移操作
        void moveWrite(size_t len)
        {
            assert((_write_idx + len) <= _buffer.size());
            _write_idx += len;
        }

    private:
        std::vector<char> _buffer;
        size_t _reader_idx; // 当前可读数据的指针---本质是下标
        size_t _write_idx;  // 当前可写数据的指针
    };
}
#endif